package com.chinasofti.IO;

import java.io.File;
import java.util.HashMap;

public class Demo05 {
    public static void main(String[] args) {
        File file = new File("E:\\Users\\apple");
        HashMap<String,Integer> hm = new HashMap<>();
        getCount(hm,file);
        System.out.println(hm);
    }
    private static void getCount(HashMap<String,Integer> hm , File file){
        File[] files = file.listFiles();//获取file下每一个文件及文件夹
        for(File f : files){
            if(f.isFile()){//判断当前f是文件吗
                //如果是文件
                String fileName = f.getName();
                String[] fileNameArr = fileName.split("\\.");
                if(fileNameArr.length == 2){
                    String fileEndName = fileNameArr[1];
                    if(hm.containsKey(fileEndName)){//如果后缀出现过则 +1并保存
                        Integer count = hm.get(fileEndName);
                        count++;
                        hm.put(fileEndName,count);
                    }else {//后缀没有出现过
                        hm.put(fileEndName,1);
                    }
                }
            }else{
                //如果是文件夹
                getCount(hm,f);
            }
        }
    }
}